package com.rundatop.core.util;

import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletResponse;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.Binarizer;
import com.google.zxing.BinaryBitmap;
import com.google.zxing.DecodeHintType;
import com.google.zxing.EncodeHintType;
import com.google.zxing.LuminanceSource;
import com.google.zxing.MultiFormatReader;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.Result;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
 
/**
 * <p>Title:QRCodeUtil </p>
 * <p>Description: 二维码生成工具类</p>
 * @author Administrator
 * @version 
 * @since 
 */
public final class QRCodeUtil extends LuminanceSource {
 
    private static final Logger logger = LoggerFactory.getLogger(QRCodeUtil.class);
 
    // 二维码颜色
    private static final int BLACK = 0xFF000000;
    // 二维码底色
    private static final int WHITE = 0xFFFFFFFF;
 
    private final BufferedImage image;
    private final int left;
    private final int top;
 
    public QRCodeUtil(BufferedImage image) {
        this(image, 0, 0, image.getWidth(), image.getHeight());
    }
 
    public QRCodeUtil(BufferedImage image, int left, int top, int width, int height) {
        super(width, height);
        int sourceWidth = image.getWidth();
        int sourceHeight = image.getHeight();
        if (left + width > sourceWidth || top + height > sourceHeight) {
            throw new IllegalArgumentException("Crop rectangle does not fit within image data.");
        }
        for (int y = top; y < top + height; y++) {
            for (int x = left; x < left + width; x++) {
                if ((image.getRGB(x, y) & 0xFF000000) == 0) {
                    image.setRGB(x, y, 0xFFFFFFFF); // = white
                }
            }
        }
        this.image = new BufferedImage(sourceWidth, sourceHeight, BufferedImage.TYPE_BYTE_GRAY);
        this.image.getGraphics().drawImage(image, 0, 0, null);
        this.left = left;
        this.top = top;
    }
 
    @Override
    public byte[] getRow(int y, byte[] row) {
        if (y < 0 || y >= getHeight()) {
            throw new IllegalArgumentException("Requested row is outside the image: " + y);
        }
        int width = getWidth();
        if (row == null || row.length < width) {
            row = new byte[width];
        }
        image.getRaster().getDataElements(left, top + y, width, 1, row);
        return row;
    }
 
    @Override
    public byte[] getMatrix() {
        int width = getWidth();
        int height = getHeight();
        int area = width * height;
        byte[] matrix = new byte[area];
        image.getRaster().getDataElements(left, top, width, height, matrix);
        return matrix;
    }
 
    @Override
    public boolean isCropSupported() {
        return true;
    }
 
    @Override
    public LuminanceSource crop(int left, int top, int width, int height) {
        return new QRCodeUtil(image, this.left + left, this.top + top, width, height);
    }
 
    @Override
    public boolean isRotateSupported() {
        return true;
    }
 
    @Override
    public LuminanceSource rotateCounterClockwise() {
        int sourceWidth = image.getWidth();
        int sourceHeight = image.getHeight();
        AffineTransform transform = new AffineTransform(0.0, -1.0, 1.0, 0.0, 0.0, sourceWidth);
        BufferedImage rotatedImage = new BufferedImage(sourceHeight, sourceWidth, BufferedImage.TYPE_BYTE_GRAY);
        Graphics2D g = rotatedImage.createGraphics();
        g.drawImage(image, transform, null);
        g.dispose();
        int width = getWidth();
        return new QRCodeUtil(rotatedImage, top, sourceWidth - (left + width), getHeight(), width);
    }

    
    /**
     * 设置源图片为背景透明，并设置透明度 返回
     * @param srcImage 源图片
     * @param alpha 透明度
     * @throws IOException
     */
    public static BufferedImage transparentImage(BufferedImage srcImage,int alpha) throws IOException {
    	int imgHeight = srcImage.getHeight();//取得图片的长和宽
    	int imgWidth = srcImage.getWidth();
    	int c = srcImage.getRGB(0,0);
    	//防止越位
    	if (alpha < 0) {
    		alpha = 0;
    	} else if (alpha > 10) {
    		alpha = 10;
    	}
    	BufferedImage bi = new BufferedImage(imgWidth, imgHeight,
    			BufferedImage.TYPE_4BYTE_ABGR);//新建一个类型支持透明的BufferedImage
    	for(int i = 0; i < imgWidth; ++i)//把原图片的内容复制到新的图片，同时把背景设为透明
    	{
    		for(int j = 0; j < imgHeight; ++j)
    		{
    			//把背景设为透明
    			if(srcImage.getRGB(i, j) == c){
    				bi.setRGB(i, j, c & 0x00ffffff);
    			}
    			//设置透明度
    			else{
    				int rgb = bi.getRGB(i, j);
    				rgb = ((alpha * 255 / 10) << 24) | (rgb & 0x00ffffff);
    				bi.setRGB(i, j, rgb);
    			}
    		}
    	}
    	return bi;
    }
 
    /**
     * 1.生成白底黑
     * @param matrix
     * @return
     */
    private static BufferedImage toBufferedImage(BitMatrix matrix) {
        int width = matrix.getWidth();
        int height = matrix.getHeight();
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        for (int x = 0; x < width; x++) {
            for (int y = 0; y < height; y++) {
        		 image.setRGB(x, y,matrix.get(x, y) ? BLACK : WHITE);
            }
        }
        return image;
    }
    
    /**
     * 2.生成透明图
     * @param matrix
     * @return
     */
    private static BufferedImage toBufferedImageTransparency(BitMatrix matrix) {
    	int width = matrix.getWidth();
    	int height = matrix.getHeight();
    	BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_4BYTE_ABGR);
    	for (int x = 0; x < width; x++) {
    		for (int y = 0; y < height; y++) {
    			boolean flag = matrix.get(x, y);
    			if (flag) {
    				image.setRGB(x, y,BLACK);
    			}
    		}
    	}
    	return image;
    }
    
    /**
     * 3.生成透明图+中间logo
     * @param matrix
     * @param logoPath
     * @param isTransparency
     * @param lx
     * @param ly
     * @return
     */
    private static BufferedImage toBufferedImageTransparency(BitMatrix matrix,String logoPath, boolean isTransparency,int lx, int ly) {
    	int width = matrix.getWidth();
    	int height = matrix.getHeight();
    	BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_4BYTE_ABGR);
    	for (int x = 0; x < width; x++) {
    		for (int y = 0; y < height; y++) {
    			boolean flag = matrix.get(x, y);
    			if (flag) {
    				image.setRGB(x, y,BLACK);
    			} else if(!isTransparency) { //是否透明标志 true 透明
    				image.setRGB(x, y,WHITE);
    			}
    		}
    	}
    	try {
    	//二维码图片
    	 Graphics2D g = image.createGraphics();
    	// 水印文件
         Image  src_biao = ImageIO.read(new File(logoPath));
         int wideth_biao = src_biao.getWidth(null);
         int height_biao = src_biao.getHeight(null);
         g.drawImage(src_biao, (width - wideth_biao) / 2 + lx, (height - height_biao) / 2 + ly, wideth_biao, height_biao, null);
         // 水印文件结束
         g.dispose();
		} catch (IOException e) {
			e.printStackTrace();
		}
    	return image;
    }
 
    /**
     * 1.生成二维码图片
     * 
     * @param matrix
     * @param format
     * @param file
     * @throws IOException
     */
    public static void writeToFile(BitMatrix matrix, String format, File file) throws IOException {
        BufferedImage image = toBufferedImage(matrix);
        if (!ImageIO.write(image, format, file)) {
            throw new IOException("Could not write an image of format " + format + " to " + file);
        }
    }
    
    /**
     * 2.生成二维码图片并写出文件（是否选择透明）
     * @param matrix
     * @param format
     * @param file
     * @param isTransparency 是否透明
     * @throws IOException
     */
    public static void writeToFile(BitMatrix matrix, String format, File file, boolean isTransparency) throws IOException {
    	BufferedImage image = isTransparency ? toBufferedImageTransparency(matrix) : toBufferedImage(matrix);
    	if (!ImageIO.write(image, format, file)) {
    		throw new IOException("Could not write an image of format " + format + " to " + file);
    	}
    }

    /**
     * 3.生成二维码图片并写出文件（是否选择透明）
     * @param matrix
     * @param format
     * @param file
     * @param logPath 中间logo地址
     * @param isTransparency 是否透明
     * @throws IOException
     */
    public static void writeToFile(BitMatrix matrix, String format, File file,String logoPath, boolean isTransparency) throws IOException {
    	BufferedImage image = toBufferedImageTransparency(matrix, logoPath,isTransparency, 0, 0);
    	if (!ImageIO.write(image, format, file)) {
    		throw new IOException("Could not write an image of format " + format + " to " + file);
    	}
    }
 
    /**
     * 生成二维码图片流
     * 
     * @param matrix
     * @param format
     * @param stream
     * @throws IOException
     */
    public static void writeToStream(BitMatrix matrix, String format, OutputStream stream) throws IOException {
        BufferedImage image = toBufferedImage(matrix);
        if (!ImageIO.write(image, format, stream)) {
            throw new IOException("Could not write an image of format " + format);
        }
    }
 
    /**
     * 1.根据内容，生成指定宽高、指定格式的二维码图片
     *
     * @param text   内容
     * @param width  宽
     * @param height 高
     * @param format 图片格式
     * @param isTransparency 是否透明
     * @return 生成的二维码图片路径
     * @throws Exception
     */
    public static String generateQRCode(String text, int width, int height, String format, String pathName, boolean isTransparency)
            throws Exception {
        Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
        hints.put(EncodeHintType.CHARACTER_SET, "utf-8");// 指定编码格式
        hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.L);// 指定纠错等级
        hints.put(EncodeHintType.MARGIN, 0); // 白边大小，取值范围0~4
        BitMatrix bitMatrix = new MultiFormatWriter().encode(text, BarcodeFormat.QR_CODE, width, height, hints);
        File outputFile = new File(pathName);
        writeToFile(bitMatrix, format, outputFile, isTransparency);
        return pathName;
    }
    /**
     * 2.根据内容，生成指定宽高、指定格式的二维码图片
     *
     * @param text   内容
     * @param width  宽
     * @param height 高
     * @param format 图片格式
     * @param isTransparency 是否透明
     * @return 生成的二维码图片路径
     * @throws Exception
     */
    public static String generateQRCode(String text, int width, int height, String format, String pathName, boolean isTransparency,String logoPath)
    		throws Exception {
    	Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
    	hints.put(EncodeHintType.CHARACTER_SET, "utf-8");// 指定编码格式
    	hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.L);// 指定纠错等级
    	hints.put(EncodeHintType.MARGIN, 0); // 白边大小，取值范围0~4
    	BitMatrix bitMatrix = new MultiFormatWriter().encode(text, BarcodeFormat.QR_CODE, width, height, hints);
    	File outputFile = new File(pathName);
    	writeToFile(bitMatrix, format, outputFile, logoPath, isTransparency);
    	return pathName;
    }
    
    /**
     * 创建二维码的统一方法(可选择是否带logo)
     * @param textUrl 要生成的二维码的文本URL
     * @param qrcodeSavePath 二维码的绝对路径
     * @param qrcodeFormat  图片后缀
     * @param size 图片大小
     * @param isLogo 是否增加logo
     * @param logoPath logo的绝对路径
     * @throws Exception
     */
    public static void CreateQRCode(String textUrl,String qrcodeSavePath,String qrcodeFormat,int size,boolean isLogo,String logoPath,boolean isTransparency) throws Exception {
		// 生成二维码图片，并返回图片路径
        QRCodeUtil.generateQRCode(textUrl, size, size, qrcodeFormat, qrcodeSavePath,isTransparency,logoPath);
    }
 
    /**
     * 输出二维码图片流
     * 
     * @param text 二维码内容
     * @param width 二维码宽
     * @param height 二维码高
     * @param format 图片格式eg: png, jpg, gif
     * @param response HttpServletResponse
     * @throws Exception
     */
    public static void generateQRCode(String text, int width, int height, String format, HttpServletResponse response)
            throws Exception {
        Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
        hints.put(EncodeHintType.CHARACTER_SET, "utf-8");// 指定编码格式
        hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.L);// 指定纠错等级
        hints.put(EncodeHintType.MARGIN, 1); // 白边大小，取值范围0~4
        BitMatrix bitMatrix = new MultiFormatWriter().encode(text, BarcodeFormat.QR_CODE, width, height, hints);
        writeToStream(bitMatrix, format, response.getOutputStream());
    }
 
    /**
     * 解析指定路径下的二维码图片
     *
     * @param filePath 二维码图片路径
     * @return
     */
    public static String parseQRCode(String filePath) {
        String content = "";
        try {
            File file = new File(filePath);
            BufferedImage image = ImageIO.read(file);
            LuminanceSource source = new QRCodeUtil(image);
            Binarizer binarizer = new HybridBinarizer(source);
            BinaryBitmap binaryBitmap = new BinaryBitmap(binarizer);
            Map<DecodeHintType, Object> hints = new HashMap<DecodeHintType, Object>();
            hints.put(DecodeHintType.CHARACTER_SET, "UTF-8");
            MultiFormatReader formatReader = new MultiFormatReader();
            Result result = formatReader.decode(binaryBitmap, hints);
 
            logger.info("result 为：" + result.toString());
            logger.info("resultFormat 为：" + result.getBarcodeFormat());
            logger.info("resultText 为：" + result.getText());
            // 设置返回值
            content = result.getText();
        } catch (Exception e) {
            logger.error(e.getMessage());
        }
        return content;
    }
 
    public static void main(String[] args) throws UnsupportedEncodingException {
        String text = "https://www.baidu.com/s?wd=%E7%A0%81%E4%BA%91&rsv_spt=1&rsv_iqid=0x964c1c3500030761&issp=1&f=8&rsv_bp=1&rsv_idx=2&ie=utf-8&rqlang=cn&tn=48021271_11_hao_pg&rsv_enter=1&oq=spring%2520data%2520jpa%25E8%2587%25AA%25E5%25AE%259A%25E4%25B9%2589%25E6%2596%25B9%25E6%25B3%2595%25E8%25A7%2584%25E8%258C%2583&rsv_t=d5f86AreI8IAKjbxCKuUMFkUO6EKFeso61joDwkdQQrpLo%2BgDuuyujzyCafz%2B5gC8IMw8TR3iRoe&rsv_pq=f35da7870001b1ba&inputT=7063&rsv_sug3=74&rsv_sug1=50&rsv_sug7=100&bs=spring%20data%20jpa%E8%87%AA%E5%AE%9A%E4%B9%89%E6%96%B9%E6%B3%95%E8%A7%84%E8%8C%83"; // 随机生成验证码
        System.out.println("随机码： " + text);
        int width = 600; // 二维码图片的宽
        int height = 600; // 二维码图片的高
        String format = "png"; // 二维码图片的格式
 
        try {
            // 生成二维码图片，并返回图片路径
            String pathName = generateQRCode(text, width, height, format, "D:/new.png",true,"F:/worktest/qrcode/static/models/logo.png");
            System.out.println("生成二维码的图片路径： " + pathName);
 
            String content = parseQRCode(pathName);
            System.out.println("解析出二维码的图片的内容为： " + content);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
 
}
